/*
 * Copyright 2021 the original author or authors.
 *
 * Licensed under the Apache, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at  http://www.gnu.org/licenses/lgpl-2.0.html
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.jn.sqlhelper.mybatis.spring.session.factory.dynamicdatasource;

import com.jn.langx.invocation.MethodInvocation;
import com.jn.sqlhelper.datasource.key.DataSourceKey;
import com.jn.sqlhelper.datasource.key.MethodInvocationDataSourceKeySelector;
import org.apache.ibatis.session.SqlSessionFactory;

public class DynamicSqlSessionFactoryMethodInvocationProvider extends DynamicDataSourceSqlSessionFactoryProvider<MethodInvocation> {

    public DynamicSqlSessionFactoryMethodInvocationProvider(DynamicSqlSessionFactory dynamicSqlSessionFactory, MethodInvocationDataSourceKeySelector keySelector) {
        setSelector(keySelector);
        setDynamicSqlSessionFactory(dynamicSqlSessionFactory);
    }

    @Override
    public SqlSessionFactory get(MethodInvocation invocation) {
        SqlSessionFactory factory = dynamicSqlSessionFactory.getDelegatingSqlSessionFactory();
        if (factory != null) {
            return factory;
        }
        DataSourceKey key = selector.select(invocation);
        return dynamicSqlSessionFactory.getDelegatingSqlSessionFactory(key);
    }
}
